package com.wwz.leetcode;


import org.junit.Test;

import java.util.Arrays;

/**
 * <p>Description:</p >
 * <p>Copyright: Copyright (c)2021</p >
 * <p>Company: Xueqiu</p >
 * <P>Created Date :2021-07-14</P>
 * <P>@version 1.0</P>
 */
//夏日炎炎，小男孩 Tony 想买一些雪糕消消暑。
//
//商店中新到 n 支雪糕，用长度为 n 的数组 costs 表示雪糕的定价，其中 costs[i] 表示第 i 支雪糕的现金价格。Tony 一共有 coins 现金可以用于消费，他想要买尽可能多的雪糕。
//
//给你价格数组 costs 和现金量 coins ，请你计算并返回 Tony 用 coins 现金能够买到的雪糕的 最大数量 。
//
//注意：Tony 可以按任意顺序购买雪糕。
//
// 
//
//示例 1：
//
//输入：costs = [1,3,2,4,1], coins = 7
//输出：4
//解释：Tony 可以买下标为 0、1、2、4 的雪糕，总价为 1 + 3 + 2 + 1 = 7
public class LeetCode1833 {

    @Test
    public void test() {
        int[] costs = new int[]{1,3,2,4,1};
        int coins = 7;
        int num = maxIceCream(costs, coins);
        System.out.println(num);
    }

    public int maxIceCream(int[] costs, int coins) {
        int rest = coins;
        int res = 0;
        int sum = 0;
        for (int cost : costs) {
            sum += cost;
        }
        if(sum == rest){
            return costs.length;
        }
        //Arrays.sort(costs);
        quickSort(costs, 0, costs.length - 1);
        for (int cost : costs) {
            if (rest - cost >= 0) {
                rest -= cost;
                res++;
            }
        }
        return res;
    }

    public void quickSort(int[] array, int left, int right) {
        if (left >= right) {
            return;
        }
        int index = partition(array, left, right);
        quickSort(array, left, index - 1);
        quickSort(array, index + 1, right);
    }

    public int partition(int[] array, int left, int right) {
        int pivot = array[left];
        while (left < right) {
            while (left < right && array[right] >= pivot) {
                right--;
            }
            array[left] = array[right];
            while (left < right && array[left] <= pivot) {
                left++;
            }
            //交换数据
            array[right] = array[left];

        }
        array[left] = pivot;
        return left;

    }

}
